package datastrucure.tree;

/**
 * @Author Fizz Pu
 * @Date 2021/4/19 下午4:34
 * @Version 1.0
 * 失之毫厘，缪之千里！
 */

/**
 * 给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
 *
 * 百度百科中最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，
 * 满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”
 *
 * 例如，给定如下二叉树:  root = [3,5,1,6,2,0,8,null,null,7,4]
 *
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

/**
 * 首先通过深度优先搜索得到链表, 然后求链表的公共交点
 */


import java.util.*;

public class Offer68_2 {

    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        ArrayList<TreeNode> listP = new ArrayList<>();
        ArrayList<TreeNode> listQ = new ArrayList<>();
        search(root, p, listP);
        search(root, q, listQ);

        TreeNode resultNode = null;
        if(!listP.isEmpty() && !listQ.isEmpty()){
            Set<TreeNode> set = new HashSet<>(listP);
            for(TreeNode node : listQ){
                if(set.contains(node)){
                    resultNode = node;
                }
            }
        }

        return resultNode;
    }

    private boolean search(TreeNode root, TreeNode targetNode, ArrayList<TreeNode> res){
        if(root == null){
            return false;
        }

        res.add(root);
        if(root == targetNode){
            return true;
        }

        boolean inLeft = false;
        boolean inRight = false;
        if(root.left != null){
            inLeft = search(root.left, targetNode, res);
        }

        if(!inLeft && root.right != null){
            inRight = search(root.right, targetNode, res);
        }

        if(!inLeft && !inRight) {
            res.remove(res.size() - 1);
        }
        return inLeft || inRight;
    }
}
